package com.demo.chatback.component;


import cn.hutool.core.date.DateUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.demo.chatback.constant.SystemConstant;
import com.demo.chatback.pojo.ro.chat.ChatInfoRO;
import com.demo.chatback.util.RedisUtil;
import org.omg.PortableInterceptor.SYSTEM_EXCEPTION;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.time.LocalDateTime;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @Description 聊天
 * @Author 小乌龟
 * @Date 2022/12/6 19:38
 */
@ServerEndpoint(value = "/chat/{userId}")
@Component
public class WebSocketServer {

    private static final Logger log = LoggerFactory.getLogger(WebSocketServer.class);

    /**
     * 记录当前在线连接数 userId session
     */
    public static final Map<String , Session> sessionMap = new ConcurrentHashMap<>();

    private static WebSocketServer webSocketServer;
    @Autowired
    private RedisUtil redisUtil;
    @PostConstruct
    public void init() {
        webSocketServer = this;
        webSocketServer.redisUtil = this.redisUtil;

    }



    /**
     * 连接建立成功调用的方法
     */
    @OnOpen
    public void onOpen(Session session, @PathParam("userId") String userId) {
            sessionMap.put(userId, session);
            log.info("有新用户加入，userId={}, 当前在线人数为：{}", userId, sessionMap.size());
    }

    /**
     * 连接关闭调用的方法
     */
    @OnClose
    public void onClose(Session session, @PathParam("userId") String userId) {
            sessionMap.remove(userId);
            log.info("有一连接关闭，移除userId={}的用户session, 当前在线人数为：{}", userId, sessionMap.size());
    }

    /**
     * 收到客户端消息后调用的方法
     * 后台收到客户端发送过来的消息
     * onMessage 是一个消息的中转站
     * 接受 浏览器端 socket.send 发送过来的 json数据
     * @param message 客户端发送过来的消息
     */
    @OnMessage
    public void onMessage(String message, Session session,@PathParam("userId") String userId) {
        //自己发送给自己消息 保持持久连接
        if (SystemConstant.CHAT_CON.equals(message)){
            try{
                log.info("服务端收到用户userid={}的消息====:{}", userId ,message);
                session.getBasicRemote().sendText(SystemConstant.CHAT_CON);
            }catch (IOException e){
                log.error(e.getMessage());
            }
            return;
        }
        JSONObject obj = JSONUtil.parseObj(message);
        // to表示发送给哪个用户，比如 admin
        String  toUserId = obj.getStr("to");
        log.info("服务端收到用户userid={}的消息====:{}", userId ,toUserId,message);
        // 发送的消息文本  hello
        String text = obj.getStr("text");
        // {"to": "admin", "text": "聊天文本"}
        // 根据 to用户名来获取 session，再通过session发送消息文本
        Session toSession =sessionMap.get(toUserId);
        if (toSession !=null){
            // 服务器端 再把消息组装一下，组装后的消息包含发送人和发送的文本内容
            // {"from": "zhang", "text": "hello"}
            JSONObject jsonObject = new JSONObject();
            // from 是 zhang
            jsonObject.set("from", userId);
            // text 同上面的text
            jsonObject.set("text", text);
            this.sendMessage(jsonObject.toString(), toSession);
            log.info("发送给用户username={}，消息：{}", toUserId, jsonObject.toString());
        }else {
            log.info("接收者未在线===存入redis中==={}=====",toUserId);
            //把发送消息者 存入缓存中 key: chat+sendUserId list: fromUserId text dateTime
            webSocketServer.redisUtil.rightPushValue(SystemConstant.REDIS_KEY_CHAT+toUserId,new ChatInfoRO().setFromUserId(userId).setText(text).setDateTime(DateUtil.formatLocalDateTime(LocalDateTime.now())));
            //发送者 用于 前端添加新聊天对象
            webSocketServer.redisUtil.sSet(SystemConstant.REDIS_KEY_SENDER+toUserId,userId);
        }
    }
    @OnError
    public void onError(Session session, Throwable error) {
        log.error("发生错误");
        error.printStackTrace();
    }

    /**
     * 服务端发送消息给客户端
     */
    private void sendMessage(String message, Session toSession) {
        try {
            log.info("服务端给客户端[{}]发送消息{}", toSession.getId(), message);
            toSession.getBasicRemote().sendText(message);
        } catch (Exception e) {
            log.error("服务端发送消息给客户端失败", e);
        }
    }

}
